package club.xiaojiawei.monotonestack;

import java.util.*;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 6/12/22 2:37 PM
 * @question 496. 下一个更大元素 I
 * @description nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
 * 给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。
 * 对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。
 * 返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。
 */
public class NextGreaterElement496 {

    public static void main(String[] args) {
        NextGreaterElement496 test = new NextGreaterElement496();
        int[] result = test.nextGreaterElement(new int[]{2, 4}, new int[]{1, 2, 3, 4});
        System.out.println(Arrays.toString(result));
    }

    /**
     * 单调栈+哈希
     * @param nums1
     * @param nums2
     * @return
     */
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int[] result = new int[nums1.length];
        int[] arr = new int[nums2.length];
        for (int i = nums2.length - 2; i >= 0; i--) {
            int j = i + 1;
            int n = nums2[i];
            while (true){
                if (nums2[i] < nums2[j]){
                    arr[i] = j - i;
                    map.put(n, nums2[j]);
                    break;
                }else if (arr[j] == 0){
                    arr[i] = 0;
                    map.put(n, -1);
                    break;
                }else if (nums2[i] == nums2[j]){
                    arr[i] = j - i + arr[j];
                    map.put(n, nums2[j]);
                    break;
                }else {
                    j += arr[j];
                }
            }
        }
        map.put(nums2[nums2.length - 1], -1);
        for (int i = 0; i < nums1.length; i++) {
            result[i] = map.get(nums1[i]);
        }
        return result;
    }

    /**
     * 官方-单调栈+哈希
     * @param nums1
     * @param nums2
     * @return
     */
    public int[] nextGreaterElement2(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        Deque<Integer> stack = new ArrayDeque<Integer>();
        for (int i = nums2.length - 1; i >= 0; --i) {
            int num = nums2[i];
            while (!stack.isEmpty() && num >= stack.peek()) {
                stack.pop();
            }
            map.put(num, stack.isEmpty() ? -1 : stack.peek());
            stack.push(num);
        }
        int[] res = new int[nums1.length];
        for (int i = 0; i < nums1.length; ++i) {
            res[i] = map.get(nums1[i]);
        }
        return res;
    }
}
